import torch
from torch import tensor

from utils.randomUtils import weightedRandomSelect

easy_indices = torch.tensor([2,4,7,9])
mean_confidences = torch.arange(start=0.1,end=2,step=0.1)

easy_confidences = mean_confidences[easy_indices]
easy_weights = 1/easy_confidences
print(easy_weights)
select_index = weightedRandomSelect(easy_weights,2)
print(select_index)
print(easy_indices[select_index])
